import pprint

import google.generativeai as genai

genai.configure(api_key="AIzaSyBJoz7BvdFgWTBwzcu-0xWpJKfEJOR6vPM")


def test_list_models():
    for model in genai.list_models():
        pprint.pprint(model)


def test_get_model():
    model = genai.get_tuned_model('models/gemini-1.5-pro-latest')
    pprint.pprint(model)


if __name__ == '__main__':
    test_get_model()
